import os
import torch
import numpy as np

# 设置环境变量以限制MKL线程数
os.environ["MKL_NUM_THREADS"] = "4"  # 限制MKL使用的线程数
os.environ["OMP_NUM_THREADS"] = "4"  # 限制OpenMP使用的线程数

# 运行训练脚本
os.system("python train.py")
